package com.zjinyin.forum.controller;

import com.sun.istack.internal.NotNull;
import com.zjinyin.forum.common.AppResult;
import com.zjinyin.forum.common.ResultCode;
import com.zjinyin.forum.config.AppConfig;
import com.zjinyin.forum.exception.ApplicationException;
import com.zjinyin.forum.model.User;
import com.zjinyin.forum.service.IUserService;
import com.zjinyin.forum.utils.MD5Util;
import com.zjinyin.forum.utils.StringUtil;
import com.zjinyin.forum.utils.UUIDUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.NonNull;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.io.File;
import java.io.IOException;
import java.util.UUID;

/**
 * Created with IntelliJ IDEA.
 * Description:
 * User: DELL
 * Date: 2023-08-09
 * Time: 15:26
 */
@Api(tags = "用户接口")
@Slf4j
@RestController
@RequestMapping("/user")
public class UserController {

    @Resource
    private IUserService userService;

    /**
     * 注册接口
     *
     * @param username       用户名
     * @param nickname       用户昵称
     * @param password       用户明文密码
     * @param passwordRepeat 用户重复密码
     *                       <p>
     *                                                                                                               TODO : 非空校验交于 lombok 的注解 @NonNull
     * @return 返回
     */
    @ApiOperation("用户注册接口")
    @PostMapping("/register")
    public AppResult register(@ApiParam("用户名") @RequestParam("username") @NonNull String username,
                              @ApiParam("用户昵称") @RequestParam("nickname") @NonNull String nickname,
                              @ApiParam("用户密码") @RequestParam("password") @NonNull String password,
                              @ApiParam("重复密码") @RequestParam("passwordRepeat") @NonNull String passwordRepeat) {
        if (!password.equals(passwordRepeat)) {
            // 打印日志
            log.warn(ResultCode.FAILED_TWO_PWD_NOT_SAME.toString());

            // 返回错误信息给前端
            return AppResult.failed(ResultCode.FAILED_TWO_PWD_NOT_SAME);
        }

        User user = new User();
        user.setUsername(username);
        user.setNickname(nickname);

        // 密码加盐加密
        String salt = UUIDUtil.UUID_32();
        String ciphertext = MD5Util.md5(password, salt);

        user.setPassword(ciphertext);
        user.setSalt(salt);

        // 调用 Service 层
        userService.createNormalUser(user);
        return AppResult.success();
    }

    /**
     * 用户登录
     *
     * @param username 用户名
     * @param password 密码
     * @param request  请求
     * @return 对密码进行加盐加密存储, 并登陆成功创建会话信息
     */
    @ApiOperation("用户登录接口")
    @PostMapping("/login")
    public AppResult login(@ApiParam("用户名") @RequestParam("username") @NotNull String username,
                           @ApiParam("用户密码") @RequestParam("password") @NonNull String password,
                           HttpServletRequest request) {

        // 1. 根据 username 获取用户
        User user = userService.login(username, password);

        // 2. 对用户进行判断
        if (user == null) {
            // 不存在这个用户名 - 提示账号或密码错误
            log.warn(ResultCode.FAILED_LOGIN.toString() + ", username : " + username);
            throw new ApplicationException(AppResult.failed(ResultCode.FAILED_LOGIN));
        }

        // 3. 登陆成功创建会话
        HttpSession session = request.getSession(true);
        session.setAttribute(AppConfig.USER_SESSION, user);
        return AppResult.success();

    }


    /**
     * 获取指定 Id 的用户信息
     *
     * @param request 请求
     * @param id      指定用户 Id
     * @return 返回制定 Id 对象
     */
    @ApiOperation("获取用户信息")
    @GetMapping("/info")
    public AppResult<User> getUserInfo(HttpServletRequest request,
                                       @ApiParam("用户 id 编号") @RequestParam(value = "id", required = false) Long id) {

        User user = null;

        // 1. 参数校验
        if (id == null) {
            // 从当前 session 中获取当前用户
            HttpSession session = request.getSession(false);

            // 已经由拦截器进行拦截, 不在需要判断
//            if (session == null || session.getAttribute(AppConfig.USER_SESSION) == null) {
//                // 当前用户未登陆
//                log.warn(ResultCode.FAILED_FORBIDDEN.toString());
//                return AppResult.failed(ResultCode.FAILED_FORBIDDEN);
//            }

            // 会话存在, 且参数正确
            user = (User) session.getAttribute(AppConfig.USER_SESSION);
            user = userService.selectByPrimaryKey(user.getId());

        } else {
            // 获取指定用户 id
            user = userService.selectByPrimaryKey(id);
        }

        // 2. 对 user 进行检验
        if (user == null) {
            log.warn(ResultCode.FAILED_USER_NOT_EXISTS.toString());
            return AppResult.failed(ResultCode.FAILED_USER_NOT_EXISTS);
        }

        // 对查询到的 user 用户进行安全处理, 屏蔽敏感信息
        // 2.1. 密码和盐值盐值屏蔽不可展示
        user.setPassword("");
        // 2.2. 创建时间和更新时间的格式化
        // 2.3  删除状态无需展示
        return AppResult.success(user);
    }

    /**
     * 退出登陆, 消除会话及其对应所有信息
     */
    @ApiOperation("退出登陆")
    @GetMapping("/logout")
    public AppResult logout(HttpServletRequest request) {

        // 1. 获取当前用户会话
        HttpSession session = request.getSession(false);

        // 2. 判断 session
        if (session != null) {
            // 销毁 session
            log.info("退出登陆成功 ");
            session.invalidate(); // 销毁 session 并解绑当前用户在的所有东西
        }

        return AppResult.success("退出登陆 ");
    }


    /**
     * 修改个人中心信息
     *
     * @param username 用户名
     * @param nickname 昵称
     * @param gender   性别
     * @param email    邮箱
     * @param phoneNum 号码
     * @param remark   简介
     */
    @ApiOperation("用户中心修改个人信息")
    @PostMapping("/modifyUserInfo")
    public AppResult modifyUserinfo(@ApiParam("用户名") @RequestParam(value = "username", required = false) String username,
                                    @ApiParam("昵称") @RequestParam(value = "nickname", required = false) String nickname,
                                    @ApiParam("性别") @RequestParam(value = "gender", required = false) Byte gender,
                                    @ApiParam("邮箱") @RequestParam(value = "email", required = false) String email,
                                    @ApiParam("手机号") @RequestParam(value = "phoneNum", required = false) String phoneNum,
                                    @ApiParam("个人简介") @RequestParam(value = "remark", required = false) String remark,
                                    HttpServletRequest request) {
        // 1. 获取当前登陆用户 Id
        HttpSession session = request.getSession(false);
        User user = (User) session.getAttribute(AppConfig.USER_SESSION);

        // 2. 检验所有参数是否都为空
        if (StringUtil.isEmpty(username) && StringUtil.isEmpty(nickname) &&
                gender == null && StringUtil.isEmpty(email) &&
                StringUtil.isEmpty(phoneNum) && StringUtil.isEmpty(remark)) {
            return AppResult.failed("请输入要修改的内容 !");
        }

        User updateUser = new User();
        // 只针对当前登录用户进行修改
        updateUser.setId(user.getId());

        if (!StringUtil.isEmpty("username")) {
            updateUser.setUsername(username);
        }

        if (!StringUtil.isEmpty("nickname")) {
            updateUser.setNickname(nickname);
        }

        if (gender != null) {
            updateUser.setGender(gender);
        }


        if (!StringUtil.isEmpty(email)) {
            updateUser.setEmail(email);
        }

        if (!StringUtil.isEmpty(phoneNum)) {
            updateUser.setPhoneNum(phoneNum);
        }

        if (!StringUtil.isEmpty(remark)) {
            updateUser.setRemark(remark);
        }

        // 3. 执行更新
        userService.modifyUserInfo(updateUser);

        // 4. 获取更新后的用户信息
        user = userService.selectByPrimaryKey(user.getId());

        // 5. 重置当前用户修改后的信息存于会话
        session.setAttribute(AppConfig.USER_SESSION, user);

        return AppResult.success(user);
    }


    /**
     * 用户中心修改密码
     *
     * @param oldPassword    旧密码
     * @param newPassword    新密码
     * @param passwordRepeat 确认密码
     */
    @ApiOperation("用户中心修改密码")
    @PostMapping("/modifyPwd")
    public AppResult modifyPassword(HttpServletRequest request,
                                    @ApiParam("旧密码") @RequestParam("oldPassword") @NotNull String oldPassword,
                                    @ApiParam("新密码") @RequestParam("newPassword") @NotNull String newPassword,
                                    @ApiParam("确认密码") @RequestParam("passwordRepeat") @NotNull String passwordRepeat) {
        // 1. 校验新密码与确认密码是否一致
        if (!newPassword.equals(passwordRepeat)) {
            log.warn(ResultCode.FAILED_TWO_PWD_NOT_SAME.toString());
            return AppResult.failed(ResultCode.FAILED_TWO_PWD_NOT_SAME);
        }

        // 2. 获取当前登录用户信息
        HttpSession session = request.getSession(false);
        User user = (User) session.getAttribute(AppConfig.USER_SESSION);

        // 3. 修改密码
        userService.modifyPassword(user.getId(), oldPassword, newPassword);

        // 4. 销毁 session 让用户重新登陆
        session.invalidate();

        return AppResult.success();
    }


    /**
     * 用户上传头像
     *
     * @param file    上传文件
     * @param request 获取用户请求
     */
    @ApiOperation("上传用户头像")
    @PostMapping("/upload")
    public AppResult upLoad(@ApiParam("用户上传文件") @RequestParam("file") @NotNull MultipartFile file,
                            HttpServletRequest request) {

        // 1. 获取当前用户
        HttpSession session = request.getSession(false);
        User user = (User) session.getAttribute(AppConfig.USER_SESSION);

        // 2. 检验用户状态
        if (user.getDeleteState() == 1) {
            return AppResult.failed(ResultCode.FAILED + " 当前用户状态异常");
        }

        // 3. 创建文件名称
        String fileName = UUID.randomUUID() +
                file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")); // 后缀名

        // 4. 设置数据库保存路径 -- 构建相对路径
//        File saveFile = new File("E:\\Javacode\\img\\pric\\" + fileName); // 本地环境
        File saveFile = new File("/root/img/pric/" + fileName); // 生产环境
        // 5. 上传文件到本地
        try {
            file.transferTo(saveFile);

            // 6. 写入数据库
            userService.upLoad("/img/pric/" + fileName, user.getId());

        } catch (IOException e) {
            e.printStackTrace();
        }

        return AppResult.success();
    }
}
